import os
import glob


def get_files_with_suffixes(directory, suffixes):
    """
    获取指定目录中具有指定后缀列表的文件列表。

    Parameters:
    - directory (str): 要搜索的目录路径。
    - suffixes (list of str): 要搜索的文件后缀名列表，例如 ['txt', 'json', 'csv']。

    Returns:
    - list of str: 匹配到的文件的绝对路径列表。

    """
    files = []
    for suffix in suffixes:
        # 构建搜索模式，匹配指定后缀名的文件
        search_pattern = os.path.join(directory, f'*.{suffix}')
        # 使用 glob 模块获取匹配的文件列表，并添加到 files 列表中
        files.extend(glob.glob(search_pattern))
    return files

# 判断文件夹或文件是否存在
def check_path(path):
    if os.path.exists(path):
        return True
    else:
        print("路径不存在")
        return False    

#  创建一层目录 
def make_dir(path):
    if not os.path.exists(path):
        os.mkdir(path)
        return True
    else:
        return False
    
# 创建多层目录
def make_dirs(path):
    if not os.path.exists(path):
        os.makedirs(path)
        return True
    else:
        return False


if __name__ == "__main__":
    img_path = "D:\\temp\\essay"
    files = get_files_with_suffixes(img_path, ['png'])
    print(files)